c65e02793a2c6ba988874e98339d48b4d73aa8a4,cdap-data-fabric/src/main/java/co/cask/cdap/data/stream/service/DistributedStreamService.java,DistributedStreamService,runOneIteration,#,188

Before Change


  protected void runOneIteration() throws Exception {
    LOG.trace("Performing heartbeat publishing in Stream service instance {}", instanceId);
    ImmutableMap.Builder<String, Long> sizes = ImmutableMap.builder();
    for (StreamSpecification streamSpec : streamMetaStore.listStreams()) {
      sizes.put(streamSpec.getName(), streamWriterSizeCollector.getTotalCollected(streamSpec.getName()));
    }
    heartbeatPublisher.sendHeartbeat(new StreamWriterHeartbeat(System.currentTimeMillis(), instanceId, sizes.build()));

After Change


    LOG.trace("Performing heartbeat publishing in Stream service instance {}", instanceId);
    ImmutableMap.Builder<String, Long> sizes = ImmutableMap.builder();
    Collection<StreamSpecification> specifications =
      streamMetaStore.listStreams().get(new NamespaceMeta.Builder().setId(Constants.DEFAULT_NAMESPACE).build());
    for (StreamSpecification streamSpec : specifications) {
      sizes.put(streamSpec.getName(), streamWriterSizeCollector.getTotalCollected(streamSpec.getName()));
    }